package cn.cymym.comm;

import java.io.UnsupportedEncodingException;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.security.SecureRandom;
import java.util.Arrays;

public class MyMd5Util {
	private static final String HEX_NUMS_STR="0123456789ABCDEF";
	 private static final Integer SALT_LENGTH = 12;
	  
	 /** 
	 * 将16进制字符串转换成字节数组 
	 * @param hex 
	 * @return 
	 */
	 public static byte[] hexStringToByte(String hex) {
	 int len = (hex.length() / 2);
	 byte[] result = new byte[len];
	 char[] hexChars = hex.toCharArray();
	 for (int i = 0; i < len; i++) {
	 int pos = i * 2;
	 result[i] = (byte) (HEX_NUMS_STR.indexOf(hexChars[pos]) << 4
	 | HEX_NUMS_STR.indexOf(hexChars[pos + 1]));
	 }
	 return result;
	 }
	 
	  
	 /**
	 * 将指定byte数组转换成16进制字符串
	 * @param b
	 * @return
	 */
	 public static String byteToHexString(byte[] b) {
	 StringBuffer hexString = new StringBuffer();
	 for (int i = 0; i < b.length; i++) {
	 String hex = Integer.toHexString(b[i] & 0xFF);
	 if (hex.length() == 1) {
	 hex = '0' + hex;
	 }
	 hexString.append(hex.toUpperCase());
	 }
	 return hexString.toString();
	 }
	  
	 /**
	 * 验证口令是否合法
	 * @param password
	 * @param passwordInDb
	 * @return
	 * @throws NoSuchAlgorithmException
	 * @throws UnsupportedEncodingException
	 */
	 public boolean validPassword(String password, String passwordInDb)
	 throws NoSuchAlgorithmException, UnsupportedEncodingException {
	 //将16进制字符串格式口令转换成字节数组
	 byte[] pwdInDb = hexStringToByte(passwordInDb);
	 //声明盐变量
	 byte[] salt = new byte[SALT_LENGTH];
	 //将盐从数据库中保存的口令字节数组中提取出来
	 System.arraycopy(pwdInDb, 0, salt, 0, SALT_LENGTH);
	 //创建消息摘要对象
	 MessageDigest md = MessageDigest.getInstance("MD5");
	 //将盐数据传入消息摘要对象
	 md.update(salt);
	 //将口令的数据传给消息摘要对象
	 md.update(password.getBytes("UTF-8"));
	 //生成输入口令的消息摘要
	 byte[] digest = md.digest();
	 //声明一个保存数据库中口令消息摘要的变量
	 byte[] digestInDb = new byte[pwdInDb.length - SALT_LENGTH];
	 //取得数据库中口令的消息摘要
	 System.arraycopy(pwdInDb, SALT_LENGTH, digestInDb, 0, digestInDb.length);
	 //比较根据输入口令生成的消息摘要和数据库中消息摘要是否相同
	 if (Arrays.equals(digest, digestInDb)) {
	 //口令正确返回口令匹配消息
	 return true;
	 } else {
	 //口令不正确返回口令不匹配消息
	 return false;
	 }
	 }
	 
	 
	 /**
	 * 获得加密后的16进制形式口令
	 * @param password
	 * @return
	 * @throws NoSuchAlgorithmException
	 * @throws UnsupportedEncodingException
	 */
	 public static String getEncryptedPwd(String password)
	 throws NoSuchAlgorithmException, UnsupportedEncodingException {
	 //声明加密后的口令数组变量
	 byte[] pwd = null;
	 //随机数生成器
	 SecureRandom random = new SecureRandom();
	 //声明盐数组变量
	 byte[] salt = new byte[SALT_LENGTH];
	 //将随机数放入盐变量中
	 random.nextBytes(salt);
	 
	 //声明消息摘要对象
	 MessageDigest md = null;
	 //创建消息摘要
	 md = MessageDigest.getInstance("MD5");
	 //将盐数据传入消息摘要对象
	 md.update(salt);
	 //将口令的数据传给消息摘要对象
	 md.update(password.getBytes("UTF-8"));
	 //获得消息摘要的字节数组
	 byte[] digest = md.digest();
	 
	 //因为要在口令的字节数组中存放盐，所以加上盐的字节长度
	 pwd = new byte[digest.length + SALT_LENGTH];
	 //将盐的字节拷贝到生成的加密口令字节数组的前12个字节，以便在验证口令时取出盐
	 System.arraycopy(salt, 0, pwd, 0, SALT_LENGTH);
	 //将消息摘要拷贝到加密口令字节数组从第13个字节开始的字节
	 System.arraycopy(digest, 0, pwd, SALT_LENGTH, digest.length);
	 //将字节数组格式加密后的口令转化为16进制字符串格式的口令
	 return byteToHexString(pwd);
	 }
}
